import numpy as np
import matplotlib.pyplot as plt
import os
import sys

# 获取当前脚本的完整路径
script_path = os.path.abspath(sys.argv[0])
# 从完整路径中获取目录
script_dir = os.path.dirname(script_path)
# 从完整路径中分离出文件名
script_name = os.path.basename(script_path)
# 使用 splitext() 函数分离文件名和扩展名
script_name_without_extension, _ = os.path.splitext(script_name)

# 创建保存图像的完整路径
save_path = os.path.join(script_dir, script_name_without_extension + ".png")

#自定义生成y的函数
def cData(n):
    a1 = np.cos(2*np.pi*n)
    b1 = np.exp(-n)
    return a1*b1
 
plt.rcParams['font.sans-serif'] = ['SimHei']    #用于显示中文字符
plt.rcParams['axes.unicode_minus'] = False  #用来显示负号ax1.axis('equal')   #保证绘制出来的是圆形
 
d1 = np.arange(0.0,6.0,0.1)     #生成一维数组序列
point = plt.plot(d1,cData(d1),'ro')     #绘制
plt.setp(point,'markersize',5) #设置数据点的大小
plt.setp(point,'markerfacecolor','g') #设置数据点的大小
plt.savefig(save_path, dpi=300)
plt.show()